#  Copyright (c) 2010
#  The Regents of the University of Michigan
#  All Rights Reserved

#  Permission is granted to use, copy, create derivative works, and
#  redistribute this software and such derivative works for any purpose,
#  so long as the name of the University of Michigan is not used in
#  any advertising or publicity pertaining to the use or distribution
#  of this software without specific, written prior authorization. If
#  the above copyright notice or any other identification of the
#  University of Michigan is included in any copy of any portion of
#  this software, then the disclaimer below must also be included.

#  This software is provided as is, without representation or warranty
#  of any kind either express or implied, including without limitation
#  the implied warranties of merchantability, fitness for a particular
#  purpose, or noninfringement.  The Regents of the University of
#  Michigan shall not be liable for any damages, including special,
#  indirect, incidental, or consequential damages, with respect to any
#  claim arising out of or in connection with the use of the software,
#  even if it has been or is hereafter advised of the possibility of
#  such damages.
import matplotlib
matplotlib.use('cairo.png')
from matplotlib import pyplot, pylab
import numpy
import scipy

import cPickle as pickle
import sys

def main(data_filename):
  counts = pickle.load(open(data_filename, 'r'))

  data, bins, junk = pyplot.hist(counts[counts.index(2):], bins=counts[-1], cumulative=False,
                    histtype='step', normed=True)

  pyplot.clf()

  print len(bins), len(data)
  pyplot.plot(bins[:-1], data)

  pyplot.xlabel('Ports for IP address')
  pyplot.xscale('log')
  pyplot.ylabel('# of IPs')
  pyplot.title('Cumulative Distribution of Ports per IP Address')
  pyplot.savefig(data_filename + '.png')

if __name__ == '__main__':
  main(*sys.argv[1:])
